package com.example.loop.util;

import java.io.File;
import java.io.FileFilter;
import java.util.Arrays;
import java.util.List;

public class FileUtils {

    /**
     * 获取某个文件夹下指定后缀的文件对象
     *
     * @param videoDirectory 读取的文件夹
     * @param videoTypeArray 需要的文件的后缀数组
     * @return 指定文件下指定后缀的文件
     */
    public static File[] getFilesBySuffix(File videoDirectory, String[] videoTypeArray) {

        return videoDirectory.listFiles(new FileFilter() {
            @Override
            public boolean accept(File file) {

                String[] split = file.getName().split("\\.");

                if (split.length > 0) {

                    List<String> videoTypeList = Arrays.asList(videoTypeArray);


                    String suffix = split[split.length - 1].toUpperCase();

                    return videoTypeList.contains(suffix);

                }

                return false;
            }
        });
    }
}
